package com.wallet.config;

import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.ConnectionFactory;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.support.converter.Jackson2JsonMessageConverter;
import org.springframework.amqp.support.converter.MessageConverter;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;

@Configuration
public class MyRabbitConfig {

//    @Autowired
    RabbitTemplate rabbitTemplate;

    @Primary
    @Bean
    public RabbitTemplate rabbitTemplate(ConnectionFactory connection){
        RabbitTemplate rabbitTemplate = new RabbitTemplate(connection);
        this.rabbitTemplate = rabbitTemplate;
        rabbitTemplate.setMessageConverter(messageConverter());
        initRabbitTemplate();
        return rabbitTemplate;
    }

   @Bean
   public MessageConverter messageConverter(){
       return new Jackson2JsonMessageConverter();
   }

    /**
     * 定制RabbitTemplate
     * 1、服务收到消息就回调
     *      1、spring.rabbitmq.publisher-confirms=true
     *      2、设置确认回调ConfirmCallback
     * 2、消息正确抵达队列进行回调
     *      1、spring.rabbitmq.publisher-returns=true
     *         spring.rabbitmq.template.mandatory=true
     * 3、消息端确认(保证每个消息被 正确消费，此时才可以brocker删除这个消息)
     *      1、默认是自动确认的，只要消息接收到，客户端会自动确认，服务端就会移除这个消息
     *      问题：
     *      我们收到很多消息，自动回复给服务器ack，只有一个消息处理成功，宕机了，发生消息丢失；
     *      消费者手动确认模式。只要我们没有明确告诉MQ，货物被签收。没有ack，消息就一直是unacked状态。即使Consumer宕机，会重新变为Ready
     *      2、如何签收：
     *          channel.basicAck(deliveryTag,false);签收；业务成功完成就应该签收
     *          channel.basicNack(deliveryTag,false,true);拒收；业务失败，拒签
     */
//   @PostConstruct //MyRabbitConfig对象创建完成以后，执行这个方法
   public void initRabbitTemplate(){
         //设置确认回调
        rabbitTemplate.setConfirmCallback(new RabbitTemplate.ConfirmCallback() {
            /**
             * 1、只要消息抵达Broker就ack=true
             * @param correlationData   当前消息的唯一关联数据(这个是消息的唯一id)
             * @param ack               消息是否成功收到
             * @param cause             失败的原因
             */
            @Override
            public void confirm(CorrelationData correlationData, boolean ack, String cause) {
                System.out.println("confirm...correlationData["+correlationData+"]==>ack["+ack+"]==>cause["+cause+"]");
            }
        });
        //设置消息抵达队列回调
       rabbitTemplate.setReturnCallback(new RabbitTemplate.ReturnCallback() {
           /**
            * 保要消息没有投递给指定的队列，就触发这个失败回调
            * @param message       投递失败的消息详细信息
            * @param replyCode     回复的状态码
            * @param replyText     回复的文本内容
            * @param exchange      当时这个消息发给哪个交换机
            * @param routingKey    当时这个消息用哪个路由键
            */
           @Override
           public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
                System.out.println("Fail...message["+message+"]==>replyCode["+replyCode+"]==>replyText["+replyText+"]==>exchange["+exchange+"]==>routingKey["+routingKey+"]");
           }
       });
   }

}
